absl-py
datasets
faiss-cpu
matplotlib
numpy
torch
transformers
triton